import torch
from torch.autograd.grad_mode import enable_grad
from torch_geometric.nn.inits import normal
from torch_geometric.nn import GCNConv, ChebConv  # noqa
import torch.nn.functional as F
from torch_geometric.datasets import *
import torch_geometric.transforms as T

from graph import ChunkGraphAlv1, GCOOGraph, CSRGraph
from helpers.normalizers import ChunkGraphGCNNormalizer
from dataset import MtxDataset
import nn.functional as mF

from gcn_models import *

import argparse


parser = argparse.ArgumentParser()
parser.add_argument("--job", choices=['single-class', 'multi-class', 'regression'], default='single-class')
parser.add_argument("--dataset")